package com.liu.tools;


import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPReply;

import java.io.IOException;
import java.io.InputStream;

/**
 * Created by wang0 on 2017/11/13.
 */
public class FtpUtil {
    /**
     * 检验指定路径的文件是否存在ftp服务器中
     *
     * @param filePath--指定绝对路径的文件
     * @param user--ftp服务器登陆用户名
     * @param passward--ftp服务器登陆密码
     * @param ip--ftp的IP地址
     * @param port--ftp的端口号
     * @return
     */

    public static boolean isFTPFileExist(String filePath, String user, String passward, String ip, int port) throws Exception {
        FTPClient ftp = new FTPClient();
        try {
            // 连接ftp服务器
            ftp.setConnectTimeout(5000);
            ftp.connect(ip, port);
            // 登陆
            ftp.login(user, passward);
            // 检验登陆操作的返回码是否正确
            if (!FTPReply.isPositiveCompletion(ftp.getReplyCode())) {
                ftp.disconnect();
                return false;
            }
            ftp.enterLocalActiveMode();
            // 设置文件类型为二进制，与ASCII有区别
            ftp.setFileType(FTP.BINARY_FILE_TYPE);
            // 设置编码格式
            ftp.setControlEncoding("GBK");

            // 提取绝对地址的目录以及文件名
            filePath = filePath.replace("ftp://" + ip + ":" + port + "/", "");
            String dir = filePath.substring(0, filePath.lastIndexOf("/"));
            String file = filePath.substring(filePath.lastIndexOf("/") + 1);
//            ftp.changeWorkingDirectory("/");
//            ftp.listFiles();
            // 进入文件所在目录，注意编码格式，以能够正确识别中文目录
            ftp.changeWorkingDirectory(new String(dir.getBytes("GBK"), FTP.DEFAULT_CONTROL_ENCODING));

            // 检验文件是否存在
            InputStream is = ftp.retrieveFileStream(new String(file.getBytes("GBK"), FTP.DEFAULT_CONTROL_ENCODING));
            if (is == null || ftp.getReplyCode() == FTPReply.FILE_UNAVAILABLE) {
                return false;
            }

            if (is != null) {
                is.close();
                ftp.completePendingCommand();
            }
            return true;
        } finally {
            if (ftp != null) {
                ftp.disconnect();
            }
        }
    }

    /**
     * Description: 向FTP服务器上传文件
     *
     * @param ip       FTP服务器hostname
     * @param port     FTP服务器端口
     * @param username FTP登录账号
     * @param password FTP登录密码
     * @param filePath FTP服务器文件存放路径。例如分日期存放：/2015/01/01。文件的路径为basePath+filePath
     * @param filename 上传到FTP服务器上的文件名
     * @param input    输入流
     * @return 成功返回true，否则返回false
     */
    public static boolean uploadFile(String ip, int port, String username, String password,
                                     String filePath, String filename, InputStream input) throws Exception {
        boolean result = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(ip, port);// 连接FTP服务器
            // 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器
            ftp.login(username, password);// 登录
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
//                throw new Exception("连接失败");
                return result;
            }
            //切换到上传目录
            if (!ftp.changeWorkingDirectory(filePath)) {
                //如果目录不存在创建目录
                String[] dirs = filePath.split("/");
                String tempPath = "";
                for (String dir : dirs) {
                    if (null == dir || "".equals(dir)) continue;
                    tempPath += "/" + dir;
                    if (!ftp.changeWorkingDirectory(tempPath)) {
                        if (!ftp.makeDirectory(tempPath)) {
                            return result;
                        } else {
                            ftp.changeWorkingDirectory(tempPath);
                        }
                    }
                }
            }
            //设置上传文件的类型为二进制类型
            ftp.setFileType(FTP.BINARY_FILE_TYPE);
            //上传文件
            if (!ftp.storeFile(filename, input)) {
                return result;
            }
            input.close();
            ftp.logout();
            result = true;
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                }
            }
        }
        return result;
    }

    public static void main(String[] args) throws Exception {
//        System.out.println(isFTPFileExist("ftp://192.168.3.244:21/20171225/11_11_01", "wang", "88888888", "192.168.3.244", 21));
//        System.out.println(isFTPFileExist("ftp://192.168.0.128:21/20171114/123456", "wang070625@163.com", "wang0a7a0a6a2a5a", "192.168.0.128", 21));

    }
}
